import os

class Config:
    SECRET_KEY = os.environ.get('SECRET_KEY') or 'your_secret_key'
    # 配置 MySQL 数据库连接，需要根据实际情况修改用户名、密码、主机和数据库名
    SQLALCHEMY_DATABASE_URI = 'mysql+mysqlconnector://root:123456@localhost/steel'
    SQLALCHEMY_TRACK_MODIFICATIONS = False

    # 预测模型相关配置
    DEVICE = "cpu"
    CHECKPOINT_PATHS = [
        os.path.join('checkpoints', 'epoch=87-step=27720.ckpt'),
        os.path.join('checkpoints', 'epoch=69-step=22050.ckpt'),
        os.path.join('checkpoints', 'epoch=76-step=24255.ckpt'),
        os.path.join('checkpoints', 'epoch=66-step=21105.ckpt')
    ]
    MEAN = [87.68971919]
    STD = [49.92222673]
    EPOCHS = 100
    BATCH_SIZE = 32


